package org.dromara.common.sensitive.core;

import cn.hutool.core.util.DesensitizedUtil;
import lombok.AllArgsConstructor;

import java.util.function.Function;


/**
 * @Author 刘武贵
 * @Date 2024/8/15 22:00
 * @Description 功能描述：脱敏策略
 */
@AllArgsConstructor
public enum SensitiveStrategy {

    /**
     * 身份证脱敏
     */
    ID_CARD(s -> DesensitizedUtil.idCardNum(s, 3, 4)),

    /**
     * 手机号脱敏
     */
    PHONE(DesensitizedUtil::mobilePhone),

    /**
     * 地址脱敏
     */
    ADDRESS(s -> DesensitizedUtil.address(s, 8)),

    /**
     * 邮箱脱敏
     */
    EMAIL(DesensitizedUtil::email),

    /**
     * 银行卡
     */
    BANK_CARD(DesensitizedUtil::bankCard);

    //可自行添加其他脱敏策略
    /**
     * 不可变的函数引用，用于实现字段的脱敏处理。
     * 该函数接受一个字符串作为输入，并返回一个脱敏后的字符串。
     */
    private final Function<String, String> desensitizer;

    /**
     * 提供对脱敏函数的访问。
     *
     * @return 一个Function对象，用于字符串的脱敏处理。
     */
    public Function<String, String> desensitizer() {
        return desensitizer;
    }
}
